# -*- coding: utf-8 -*-
# @Time    : 2021/7/7 14:46
# @Author  : LuoTianHang

# ####################datalist.py 说明##########################
# this script is used for the train.py to prepare the data

from torchvision.datasets import MNIST
from torchvision import transforms

trans_img = transforms.Compose([transforms.ToTensor()])
train_set=MNIST("./data/MNIST_DATA",train=True,transform=trans_img,download=True)

test_set=MNIST("./data/MNIST_DATA",train=False,transform=trans_img,download=True)
